ValueError: operands could not be broadcast together with shapes (2 1000) (2 )